package net.za.cair.dip;

import org.semanticweb.owlapi.model.OWLAxiom;
import org.semanticweb.owlapi.model.OWLException;
import org.semanticweb.owlapi.reasoner.OWLReasonerFactory;

import net.za.cair.dip.model.MaterializationRanking;
import net.za.cair.dip.model.OntologyStructure;
import net.za.cair.dip.model.Ranking;
import net.za.cair.dip.model.ReasoningType;
import net.za.cair.dip.transform.LexicographicRankingAlgorithm;
import net.za.cair.dip.transform.RationalRankingAlgorithm;
import net.za.cair.dip.util.AxiomManipulator;

/*
 * Copyright (C) 2011, Centre for Artificial Intelligence Research
 *
 * Modifications to the initial code base are copyright of their
 * respective authors, or their employers as appropriate.  Authorship
 * of the modifications may be determined from the ChangeLog placed at
 * the end of this file.
 *
 * This library is free software; you can redistribute it and/or
 * modify it under the terms of the GNU Lesser General Public
 * License as published by the Free Software Foundation; either
 * version 2.1 of the License, or (at your option) any later version.

 * This library is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
 * Lesser General Public License for more details.

 * You should have received a copy of the GNU Lesser General Public
 * License along with this library; if not, write to the Free Software
 * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA  02110-1301 USA
 */

/**
 * Author: Kody Moodley<br>
 * Centre for Artificial Intelligence Research<br>
 * UKZN and CSIR<br>
 * Date: 10-Oct-2011<br><br>
 */

public class RankingConstruction {

	private MaterializationRanking ranking;
	private Ranking displayableRanking;
	
	public RankingConstruction(OWLAxiom query, OWLReasonerFactory reasonerFactory, OntologyStructure ontology, ReasoningType algorithm) throws OWLException{
		if (algorithm.equals(ReasoningType.RATIONAL)){
			RationalRankingAlgorithm alg = new RationalRankingAlgorithm(reasonerFactory, ontology);
			ranking = alg.computeRanking();
			displayableRanking = alg.getDisplayableRanking();

		}
		else if (algorithm.equals(ReasoningType.PREFERENTIAL)){
			ontology.addAxiom(AxiomManipulator.getPreferentialAdditionAxiom(query));
			RationalRankingAlgorithm alg = new RationalRankingAlgorithm(reasonerFactory, ontology);
			ranking = alg.computeRanking();
			displayableRanking = alg.getDisplayableRanking();
		}
		else if (algorithm.equals(ReasoningType.LEXICOGRAPHIC)){
			LexicographicRankingAlgorithm alg = new LexicographicRankingAlgorithm(reasonerFactory, ontology);
			ranking = alg.computeRanking();
			displayableRanking = alg.getDisplayableRanking();
		}
		else{
			displayableRanking = new Ranking();
			ranking = new MaterializationRanking();
		}
	}
	
	public Ranking getDisplayableRanking(){
		return displayableRanking;
	}
	
	public MaterializationRanking getRanking(){
		return ranking;
	}
}
